//
// Created by xujingyi on 2021/5/15.
//

#ifndef ALLIANCE_SENTRY_V2_0_CNN_H
#define ALLIANCE_SENTRY_V2_0_CNN_H

#include<opencv2/dnn.hpp>
namespace rm {
    extern std::map<int, int> index2name;
    class CNN {
    public:
//        CNN(rm::Setter &setter);
        CNN();
        ~CNN()=default;
        int forward(const cv::Mat &input,bool is_highland_blind_view);
        cv::dnn::Net net;
    private:
        float _not_one_conf=0.75;
        float _one_conf=0.65;
        float _blind_not_one_conf=0.10;
        float _blind_one_conf=0.10;
    };
}
#endif //ALLIANCE_SENTRY_V2_0_CNN_H
